from trainv7 import *

if __name__ == "__main__":
    torch.cuda.empty_cache()
    opt = parse_opt()
    opt.weight = './weights/yolov7-tiny.pt'
    opt.data = './cjc/data/age.yaml'
    opt.img_size = [640, 640]
    opt.batch_size = 4
    opt.resume = True
    opt.name = 'age'
    try:
        main(opt)
    except KeyboardInterrupt as e:
        print("手动退出")
